cd ..
python3 train.py --img 640 --batch 8 --epoch 300 --data ./data/Custom.yaml --cfg ./models/Custom_s.yaml --weights yolov5s.pt --device '0'